import numpy as np


idxs = np.arange(0, 10)
print(idxs)
idxs_1 = idxs[None, :]
print(idxs_1)
idxs_2 = (idxs + 1 - idxs % 2 * 2)[:, None]
print(idxs_2)
labels = np.equal(idxs_1, idxs_2).astype(np.int)
print(labels)
